package com.hsu.yupao.controller;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.hsu.yupao.common.BaseResponse;
import com.hsu.yupao.common.DeleteRequest;
import com.hsu.yupao.common.ErrorCode;
import com.hsu.yupao.common.ResultUtils;
import com.hsu.yupao.exception.BusinessException;
import com.hsu.yupao.model.domain.Team;
import com.hsu.yupao.model.domain.User;
import com.hsu.yupao.model.domain.UserTeam;
import com.hsu.yupao.model.dto.TeamQuery;
import com.hsu.yupao.model.request.TeamAddRequest;
import com.hsu.yupao.model.request.TeamJoinRequest;
import com.hsu.yupao.model.request.TeamQuitRequest;
import com.hsu.yupao.model.request.TeamUpdateRequest;
import com.hsu.yupao.model.vo.TeamUserVO;
import com.hsu.yupao.service.TeamService;
import com.hsu.yupao.service.UserService;
import com.hsu.yupao.service.UserTeamService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeanUtils;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * 队伍控制器。
 * 处理队伍相关的 HTTP 请求，包括增删改查操作。
 * @ClassName: TeamController
 * @author Hsu琛君珩
 * @date 2024-05-21
 * @apiNote
 * @Version: v1.0
 */
@RestController
@RequestMapping("/team")
@CrossOrigin(origins = {"http://localhost:5173/"}, allowCredentials = "true")
@Slf4j
public class TeamController {

    @Resource
    private UserService userService;

    @Resource
    private TeamService teamService;

    @Resource
    private UserTeamService userTeamService;

    /**
     * 添加队伍接口。
     * 通过HTTP POST请求向服务器提交队伍信息，包括队伍名称、描述、最大人数、过期时间、用户ID、状态和密码等信息。
     * 首先校验请求体中的队伍信息，若为空则抛出空对象异常。
     * 然后从HTTP请求中获取当前登录用户信息，以便进行权限校验和关联队伍操作。
     * 创建一个新的队伍对象，并通过BeanUtils工具类将请求体中的数据拷贝到队伍对象中。
     * 调用队伍服务的添加队伍方法，将队伍信息和当前登录用户传入，并获取返回的队伍ID。
     * 将队伍ID包装在统一响应结构中并返回给客户端。
     * @param teamAddRequest 添加队伍请求体，包含待添加队伍的信息
     * @param request        HTTP请求对象，用于获取当前登录用户信息和进行权限校验
     * @return 返回包含添加的队伍ID的统一响应结构
     */
    @PostMapping("/add")
    public BaseResponse<Long> addTeam(@RequestBody TeamAddRequest teamAddRequest, HttpServletRequest request) {
        // 校验请求体中的队伍信息是否为空，若为空则抛出空对象异常
        if (teamAddRequest == null) {
            throw new BusinessException(ErrorCode.PARAMS_ERROR);
        }
        // 从HTTP请求中获取当前登录用户信息
        User logininUser = userService.getLoginUser(request);
        // 创建一个新的队伍对象
        Team team = new Team();
        // 将请求体中的数据拷贝到队伍对象中
        BeanUtils.copyProperties(teamAddRequest, team);
        // 调用队伍服务的添加队伍方法，将队伍信息和当前登录用户传入，并获取返回的队伍ID
        long teamId = teamService.addTeam(team, logininUser);
        // 将队伍ID包装在统一响应结构中并返回给客户端
        return ResultUtils.success(teamId);
    }

//    /**
//     * 删除队伍接口。
//     * 从数据库中删除指定 ID 的队伍记录。
//     * @param id 待删除队伍的 ID。
//     * @return 包含删除结果的统一响应结构。
//     * @throws BusinessException 当请求参数异常或删除失败时抛出业务异常。
//     */
//    @PostMapping("/delete")
//    public BaseResponse<Boolean> deleteTeam(@RequestBody long id) {
//        // 参数校验
//        if (id <= 0) {
//            throw new BusinessException(ErrorCode.PARAMS_ERROR);
//        }
//        // 执行删除操作
//        boolean result = teamService.removeById(id);
//        if (!result) {
//            throw new BusinessException(ErrorCode.SYSTEM_ERROR, "删除失败");
//        }
//        return ResultUtils.success(true);
//    }

    /**
     * 更新队伍接口。
     * 用于更新数据库中指定队伍的信息。
     * @param teamUpdateRequest 包含更新信息的队伍对象。
     * @param request HTTP请求对象，用于获取当前请求的相关信息。
     * @return 包含更新结果的统一响应结构。
     * @throws BusinessException 当请求参数异常或更新失败时抛出业务异常。
     */
    @PostMapping("/update")
    public BaseResponse<Boolean> updateTeam(@RequestBody TeamUpdateRequest teamUpdateRequest, HttpServletRequest request) {
        // 参数校验
        if (teamUpdateRequest == null) {
            throw new BusinessException(ErrorCode.PARAMS_ERROR);
        }
        User loginUser = userService.getLoginUser(request);
        boolean result = teamService.updateTeam(teamUpdateRequest, loginUser);
        if (!result) {
            throw new BusinessException(ErrorCode.SYSTEM_ERROR, "更新失败");
        }
        return ResultUtils.success(true);
    }

    /**
     * 根据 ID 获取队伍信息接口。
     * 从数据库中获取指定 ID 的队伍信息。
     * @param id 待查询队伍的 ID。
     * @return 包含查询结果的统一响应结构。
     * @throws BusinessException 当请求参数异常或查询结果为空时抛出业务异常。
     */
    @GetMapping("/get")
    public BaseResponse<Team> getTeamById(long id) {
        // 参数校验
        if (id <= 0) {
            throw new BusinessException(ErrorCode.PARAMS_ERROR);
        }
        // 查询指定 ID 的队伍信息
        Team team = teamService.getById(id);
        if (team == null) {
            throw new BusinessException(ErrorCode.NULL_ERROR);
        }
        return ResultUtils.success(team);
    }

    /**
     * 查询队伍列表接口。
     * 根据用户提供的查询条件，查询并返回符合条件的队伍列表，并标注当前登录用户是否已加入这些队伍。
     * @param teamQuery 查询条件。
     * @param request HTTP请求对象，用于获取当前请求的用户信息。
     * @return 包含查询结果和加入状态的统一响应结构。
     * @throws BusinessException 当请求参数异常或查询过程中出现问题时抛出。
     */
    @GetMapping("/list")
    public BaseResponse<List<TeamUserVO>> listTeams(TeamQuery teamQuery, HttpServletRequest request) {
        // 参数校验，确保查询条件不为空
        if (teamQuery == null) {
            throw new BusinessException(ErrorCode.PARAMS_ERROR);
        }
        // 确定当前用户是否拥有管理员权限
        boolean isAdmin = userService.isAdmin(request);
        // 1.使用提供的查询条件获取队伍列表，考虑用户是否为管理员
        List<TeamUserVO> teamList = teamService.listTeams(teamQuery, isAdmin);
        // 提取查询结果中所有队伍的ID
        final List<Long> teamIdList = teamList.stream().map(TeamUserVO::getId).collect(Collectors.toList());
        // 2.为了确定当前用户是否已加入这些队伍，查询相关联的UserTeam信息
        QueryWrapper<UserTeam> userTeamQueryWrapper = new QueryWrapper<>();
        try {
            User loginUser = userService.getLoginUser(request);
            userTeamQueryWrapper.eq("userId", loginUser.getId());
            userTeamQueryWrapper.in("teamId", teamIdList);
            List<UserTeam> userTeamList = userTeamService.list(userTeamQueryWrapper);
            // 构建已加入队伍的ID集合
            Set<Long> hasJoinTeamIdSet = userTeamList.stream().map(UserTeam::getTeamId).collect(Collectors.toSet());
            // 更新队伍列表，标注当前用户是否已加入各队伍
            teamList.forEach(team -> team.setHasJoin(hasJoinTeamIdSet.contains(team.getId())));
        } catch (Exception e) {
            // 异常处理逻辑，记录日志或其他处理
        }
        // 3.根据队伍ID，查询所有相关联的UserTeam记录，用于计算每个队伍的成员数量
        QueryWrapper<UserTeam> userTeamJoinQueryWrapper = new QueryWrapper<>();
        userTeamJoinQueryWrapper.in("teamId", teamIdList);
        List<UserTeam> userTeamJoinList = userTeamService.list(userTeamJoinQueryWrapper);
        // 将队伍成员按队伍ID分组，便于计算每个队伍的成员数量
        Map<Long, List<UserTeam>> teamIdUserTeamMap = userTeamJoinList.stream().collect(Collectors.groupingBy(UserTeam::getTeamId));
        teamList.forEach(team -> {
            // 设置每个队伍的成员数量
            team.setHasJoinNum(teamIdUserTeamMap.getOrDefault(team.getId(), new ArrayList<>()).size());
        });
        // 返回最终的队伍列表，包含成员加入情况
        return ResultUtils.success(teamList);
    }

    /**
     * 分页查询队伍列表接口。
     * 根据条件分页查询数据库中符合条件的队伍列表。
     * @param teamQuery 查询条件。
     * @return 包含查询结果的统一响应结构。
     * @throws BusinessException 当请求参数异常时抛出业务异常。
     */
    @GetMapping("/list/page")
    public BaseResponse<Page<Team>> listTeamsByPage(TeamQuery teamQuery) {
        // 参数校验
        if (teamQuery == null) {
            throw new BusinessException(ErrorCode.PARAMS_ERROR);
        }
        // 构造查询条件
        Team team = new Team();
        BeanUtils.copyProperties(teamQuery, team);
        Page<Team> page = new Page<>(teamQuery.getPageNum(), teamQuery.getPageSize());
        QueryWrapper<Team> queryWrapper = new QueryWrapper<>(team);
        // 执行分页查询操作
        Page<Team> resultPage = teamService.page(page, queryWrapper);
        return ResultUtils.success(resultPage);
    }

    /**
     * 加入队伍接口。
     * 用于处理用户的加入队伍请求，根据提供的队伍加入信息，将登录用户添加到指定的队伍中。
     * @param teamJoinRequest 包含队伍加入信息的请求对象。
     * @param request HTTP请求对象，用于获取当前请求的用户信息。
     * @return 包含加入操作结果的统一响应结构。返回true表示加入成功，false表示加入失败。
     * @throws BusinessException 当请求参数异常或加入操作失败时抛出业务异常。
     */
    @PostMapping("/join")
    public BaseResponse<Boolean> joinTeam(@RequestBody TeamJoinRequest teamJoinRequest, HttpServletRequest request) {
        // 参数校验，确保传入的请求体不为空
        if (teamJoinRequest == null) {
            throw new BusinessException(ErrorCode.PARAMS_ERROR, "请求体不能为空");
        }
        // 从HTTP请求中获取当前登录的用户信息
        User loginUser = userService.getLoginUser(request);
        if (loginUser == null) {
            throw new BusinessException(ErrorCode.NOT_LOGIN, "用户未登录");
        }
        // 调用服务层方法处理加入队伍的请求
        boolean result = teamService.joinTeam(teamJoinRequest, loginUser);
        // 根据服务层返回的结果，构造响应体
        if (!result) {
            throw new BusinessException(ErrorCode.SYSTEM_ERROR, "加入队伍失败");
        }
        // 如果加入成功，返回成功的响应
        return ResultUtils.success(result);
    }

    /**
     * 处理用户退出队伍的请求。
     * 该接口接收客户端提交的退出队伍请求，验证请求的合法性，并调用服务层执行退出操作。
     *
     * @param teamQuitRequest 包含队伍ID的请求体，用于指定用户想要退出的队伍。
     * @param request HTTP请求对象，用于从中获取当前登录用户的信息。
     * @return 返回一个包含操作结果的统一响应结构，成功时返回true，失败时返回false。
     * @throws BusinessException 当请求体为null或者其他业务校验失败时，抛出业务异常。
     */
    @PostMapping("/quit")
    public BaseResponse<Boolean> quitTeam(@RequestBody TeamQuitRequest teamQuitRequest, HttpServletRequest request) {
        // 校验请求体是否为空
        if (teamQuitRequest == null) {
            throw new BusinessException(ErrorCode.PARAMS_ERROR, "请求体不能为空");
        }
        // 从会话中获取当前登录的用户信息
        User loginUser = userService.getLoginUser(request);
        // 调用业务层执行退出队伍的操作
        boolean result = teamService.quitTeam(teamQuitRequest, loginUser);
        // 根据业务层返回的结果，生成并返回响应体
        return ResultUtils.success(result);
    }

    /**
     * 删除队伍接口。
     * 用于根据传入的请求删除指定的队伍。该方法首先验证请求参数的有效性，然后根据当前登录用户的权限（例如管理员或队伍创建者）
     * 判断是否有权执行删除操作。只有验证成功的用户才能删除队伍。
     * @param deleteRequest 包含要删除队伍ID的请求对象。
     * @param request HTTP请求对象，用于获取当前请求的用户会话信息，以进行身份验证和权限检查。
     * @return 包含操作成功或失败的响应。如果删除成功，则返回true，如果失败则返回false，并可能抛出异常说明原因。
     * @throws BusinessException 如果提供的队伍ID无效、用户无删除权限或删除过程中发生其他错误，则抛出业务异常。
     */
    @PostMapping("/delete")
    public BaseResponse<Boolean> deleteTeam(@RequestBody DeleteRequest deleteRequest, HttpServletRequest request) {
        // 检查请求对象及ID的有效性
        if (deleteRequest == null || deleteRequest.getId() <= 0) {
            throw new BusinessException(ErrorCode.PARAMS_ERROR);
        }
        long id = deleteRequest.getId();
        // 获取当前登录用户信息
        User loginUser = userService.getLoginUser(request);
        // 执行删除操作
        boolean result = teamService.deleteTeam(id, loginUser);
        if (!result) {
            throw new BusinessException(ErrorCode.SYSTEM_ERROR, "删除失败");
        }
        // 返回操作结果
        return ResultUtils.success(result);
    }

    /**
     * 获取当前用户创建的队伍列表。
     * 此接口根据当前登录用户的 ID 和提供的查询条件来检索由该用户创建的队伍列表。
     * 主要用于显示用户自己创建的所有队伍信息。
     * @param teamQuery 查询条件对象，包含可筛选的各种字段。
     * @param request HTTP请求对象，用于获取当前请求的用户会话信息。
     * @return 返回统一格式的响应，包含符合条件的队伍列表。
     * @throws BusinessException 如果请求参数为 null 或其他业务规则不满足时抛出。
     */
    @GetMapping("/list/my/create")
    public BaseResponse<List<TeamUserVO>> listMyCreateTeams(TeamQuery teamQuery, HttpServletRequest request) {
        // 参数校验，确保传入的查询条件不为null
        if (teamQuery == null) {
            throw new BusinessException(ErrorCode.PARAMS_ERROR);
        }
        // 从会话中获取当前登录的用户信息
        User loginUser = userService.getLoginUser(request);
        // 设置查询条件中的用户ID为当前登录用户的ID，以确保只查询当前用户创建的队伍
        teamQuery.setUserId(loginUser.getId());
        // 调用服务层方法，传入查询条件，获取符合条件的队伍列表
        List<TeamUserVO> teamList = teamService.listTeams(teamQuery, true);
        // 封装并返回查询结果
        return ResultUtils.success(teamList);
    }

    /**
     * 获取当前用户加入的队伍列表。
     * 此接口根据当前登录用户的 ID 来检索其加入的所有队伍。
     * @param teamQuery 查询条件，可以包括各种筛选字段，但主要是通过用户加入的队伍ID进行过滤。
     * @param request HTTP请求对象，用于获取当前请求的用户会话。
     * @return 返回当前用户加入的队伍列表的统一响应结构。
     * @throws BusinessException 当请求参数为 null 或其他业务规则不满足时抛出。
     */
    @GetMapping("/list/my/join")
    public BaseResponse<List<TeamUserVO>> listMyJoinTeams(TeamQuery teamQuery, HttpServletRequest request) {
        // 参数校验，确保传入的查询条件不为null
        if (teamQuery == null) {
            throw new BusinessException(ErrorCode.PARAMS_ERROR);
        }
        // 从会话中获取当前登录的用户信息
        User loginUser = userService.getLoginUser(request);
        // 构造查询条件，查询当前用户加入的所有队伍
        QueryWrapper<UserTeam> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("userId", loginUser.getId());
        // 获取当前用户加入的队伍的映射列表
        List<UserTeam> userTeamList = userTeamService.list(queryWrapper);
        // 从数据列表中提取不重复的队伍ID并关联对应的用户ID
        // 示例数据:
        //  | teamId | userId |
        //  |--------|--------|
        //  |   1    |   2    |
        //  |   1    |   3    |
        //  |   2    |   3    |
        // 经过处理后的结果为一个映射（Map），其中每个队伍ID键对应一个包含所有相关联用户ID的列表：
        //  | teamId | userIds   |
        //  |--------|-----------|
        //  |   1    | [2, 3]    |  // 队伍1关联了用户2和用户3
        //  |   2    | [3]       |  // 队伍2只关联了用户3
        Map<Long, List<UserTeam>> listMap = userTeamList.stream().collect(Collectors.groupingBy(UserTeam::getTeamId));
        // 获取所有队伍ID，设置为查询条件
        ArrayList<Long> idList = new ArrayList<>(listMap.keySet());
        teamQuery.setIdList(idList);
        // 无需管理员权限即可查询，因此传入true
        List<TeamUserVO> teamList = teamService.listTeams(teamQuery, true);
        // 封装并返回查询结果
        return ResultUtils.success(teamList);
    }

}
